package cn.iocoder.yudao.module.product.dal.mysql.sku;

import cn.hutool.core.lang.Assert;
import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.iocoder.yudao.module.product.dal.dataobject.sku.ProductSkuDO;
import cn.iocoder.yudao.module.product.dal.dataobject.sku.ProductSkuDenominationDO;
import cn.iocoder.yudao.module.product.dal.dataobject.sku.SkuDiscountDTO;
import cn.iocoder.yudao.module.product.enums.spu.ProductSkuStatusEnum;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;

import java.time.LocalDateTime;
import java.util.Collection;
import java.util.List;
import java.util.Set;

@Mapper
public interface ProductSkuMapper extends BaseMapperX<ProductSkuDO> {

    default List<ProductSkuDO> selectListBySpuId(Long spuId) {
        return selectList(new LambdaQueryWrapperX<ProductSkuDO>().eq(ProductSkuDO::getSpuId, spuId).orderByDesc(ProductSkuDO::getSort));
    }

    default List<ProductSkuDO> selectListBySpuIdAndSkuIds(Long spuId, Set<Long> skuIds) {
        return selectList(new LambdaQueryWrapperX<ProductSkuDO>()
                .eq(ProductSkuDO::getSpuId, spuId)
                .inIfPresent(ProductSkuDO::getId, skuIds)
                .orderByAsc(ProductSkuDO::getSort));
    }

    default List<ProductSkuDO> selectListBySpuId(Collection<Long> spuIds) {
        return selectList(ProductSkuDO::getSpuId, spuIds);
    }

    default List<ProductSkuDO> selectListBySpuIdAndStatus(ProductSkuStatusEnum statusEnum, Collection<Long> spuIds) {
        return selectList(new LambdaQueryWrapper<ProductSkuDO>()
                .in(ProductSkuDO::getSpuId, spuIds)
                .eq(ProductSkuDO::getStatus, statusEnum.getStatus())
                .orderByAsc(ProductSkuDO::getSpuId)
                .orderByAsc(ProductSkuDO::getSort)
        );
    }

    default void deleteBySpuId(Long spuId) {
        delete(new LambdaQueryWrapperX<ProductSkuDO>().eq(ProductSkuDO::getSpuId, spuId));
    }

    /**
     * 更新 SKU 库存（增加）
     *
     * @param id        编号
     * @param incrCount 增加库存（正数）
     */
    default void updateStockIncr(Long id, Integer incrCount) {
        Assert.isTrue(incrCount > 0);
        LambdaUpdateWrapper<ProductSkuDO> lambdaUpdateWrapper = new LambdaUpdateWrapper<ProductSkuDO>()
                .setSql(" stock = stock + " + incrCount)
                .eq(ProductSkuDO::getId, id);
        update(null, lambdaUpdateWrapper);
    }

    /**
     * 更新 SKU 库存（减少）
     *
     * @param id        编号
     * @param incrCount 减少库存（负数）
     * @return 更新条数
     */
    default int updateStockDecr(Long id, Integer incrCount) {
        Assert.isTrue(incrCount < 0);
        LambdaUpdateWrapper<ProductSkuDO> updateWrapper = new LambdaUpdateWrapper<ProductSkuDO>()
                .setSql(" stock = stock + " + incrCount) // 负数，所以使用 + 号
                .eq(ProductSkuDO::getId, id)
                .ge(ProductSkuDO::getStock, -incrCount); // cas 逻辑
        return update(null, updateWrapper);
    }

    default List<ProductSkuDO> searchSpuByDataAllowanceAndValidityPeriodDays(Set<Long> spuIdSet,
                                                                             Integer esDataAllowanceMb,
                                                                             Integer esValidityPeriodDaysMin,
                                                                             Integer esValidityPeriodDaysMax) {
        return selectList(new LambdaQueryWrapperX<ProductSkuDO>()
                .in(ProductSkuDO::getSpuId, spuIdSet)
                .eq(ProductSkuDO::getEsDataAllowanceMb, esDataAllowanceMb)
                .ge(ProductSkuDO::getEsValidityPeriodDays, esValidityPeriodDaysMin)
                .le(ProductSkuDO::getEsValidityPeriodDays, esValidityPeriodDaysMax)
        );
    }

    default ProductSkuDO getSkuByEsProductId(String esProductId) {
        return selectOne(ProductSkuDO::getEsProductId, esProductId);
    }


    default List<ProductSkuDO> getSkuByCardKeyProductNo(String productNo) {
        return selectList(new LambdaQueryWrapperX<ProductSkuDO>()
                .eq(ProductSkuDO::getCardKeyProductNo, productNo)
        );
    }

    List<ProductSkuDenominationDO> selectMinDenominationSkuBySpuIds(@Param("spuIds") List<Long> spuIds);

    List<SkuDiscountDTO> findMaxDiscountSkuBySpuIds(@Param("spuIds") List<Long> spuIds, @Param("currentTime") LocalDateTime currentTime);

}
